package com.system.ad.mysql;

import com.alibaba.fastjson.JSON;
import com.system.ad.mysql.constant.OpType;
import com.system.ad.mysql.dto.ParseTemplate;
import com.system.ad.mysql.dto.TableTemplate;
import com.system.ad.mysql.dto.Template;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.List;
import java.util.Map;

/**
 * 模板管理类，用于加载模板信息和元数据信息
 */
@Slf4j
@Component
public class TemplateHolder {

    private ParseTemplate template;
    private final JdbcTemplate jdbcTemplate;

    private String SQL_SCHEMA = "select table_schema, table_name, " +
            "column_name, ordinal_position from information_schema.columns " +
            "where table_schema = ? and table_name = ?";

    /**
     * 构造方法，注入 JdbcTemplate 对象
     *
     * @param jdbcTemplate JdbcTemplate 对象
     */
    @Autowired
    public TemplateHolder(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
    }

    /**
     * 初始化方法，在构造完成后调用，加载模板信息
     */
    @PostConstruct
    private void init() {
        loadJson("template.json");
    }

    /**
     * 根据表名获取表格模板
     *
     * @param tableName 表名
     * @return 表格模板对象
     */
    public TableTemplate getTable(String tableName) {
        return template.getTableTemplateMap().get(tableName);
    }

    /**
     * 加载 JSON 文件并解析为模板信息
     *
     * @param path JSON 文件路径
     */
    private void loadJson(String path) {
        // 获取当前线程的类加载器
        ClassLoader cl = Thread.currentThread().getContextClassLoader();
        // 读取 JSON 文件流
        InputStream inStream = cl.getResourceAsStream(path);

        try {
            // 解析 JSON 文件为模板对象
            Template template = JSON.parseObject(
                    inStream,
                    Charset.defaultCharset(),
                    Template.class
            );
            // 解析模板对象并存储到成员变量中
            this.template = ParseTemplate.parse(template);
            // 加载元数据信息
            loadMeta();
        } catch (IOException ex) {
            log.error(ex.getMessage());
            // 抛出运行时异常，表示解析 JSON 文件失败
            throw new RuntimeException("fail to parse json file");
        }
    }

    /**
     * 加载元数据信息并存储到表格模板对象中
     */
    private void loadMeta() {
        // 遍历模板中的每个表格模板
        for (Map.Entry<String, TableTemplate> entry :
                template.getTableTemplateMap().entrySet()) {

            TableTemplate table = entry.getValue(); // 获取当前表格模板

            // 获取更新、添加、删除操作的字段集合
            List<String> updateFields = table.getOpTypeFieldSetMap().get(
                    OpType.UPDATE
            );
            List<String> insertFields = table.getOpTypeFieldSetMap().get(
                    OpType.ADD
            );
            List<String> deleteFields = table.getOpTypeFieldSetMap().get(
                    OpType.DELETE
            );

            // 查询数据库元数据信息
            jdbcTemplate.query(SQL_SCHEMA, new Object[]{
                    template.getDatabase(), table.getTableName()
            }, (rs, i) -> {

                int pos = rs.getInt("ORDINAL_POSITION"); // 获取列的位置
                String colName = rs.getString("COLUMN_NAME"); // 获取列名

                // 判断列是否属于更新、添加或删除操作的字段集合之一
                if ((updateFields != null && updateFields.contains(colName))
                        || (insertFields != null && insertFields.contains(colName))
                        || (deleteFields != null && deleteFields.contains(colName))) {
                    // 将列的位置和名称存储到表格模板的位置映射中
                    table.getPosMap().put(pos - 1, colName);
                }

                return null;
            });
        }
    }
}
